import socket


def main():
    # 创建一个tcp套接字
    tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 创建连接
    tcp_socket.connect(("127.0.0.1", 8888))

    while True:
        file_name = input("请输入要下载的文件\n")
        tcp_socket.send(file_name.encode("utf-8"))
        # 接受文件的数据
        recv = tcp_socket.recv(1024 * 1024)

        # 接受的数据保存到文件中, wb写二进制
        # with文件打得开的前提下, 出现异常可以关闭socket
        with open("py_005_c_file.txt", "wb") as f:
            f.write(recv)

    # 关闭套接字
    tcp_socket.close()


# 程序入口
if __name__ == "__main__":
    main()
